// Copyright 2019 syncd Author. All Rights Reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package syncd

import (
	"errors"
	"fmt"
	"time"

	_ "github.com/go-sql-driver/mysql"
	"github.com/jinzhu/gorm"
)

type DB struct {
	DbHandler *gorm.DB
	cfg       *DbConfig
}

func NewDatabase(cfg *DbConfig) *DB {
	return &DB{
		cfg: cfg,
	}
}

func (db *DB) Open() error {
	c, err := gorm.Open("mysql", db.parseConnConfig())
	if err != nil {
		return errors.New(fmt.Sprintf("mysql connect failed, %s", err.Error()))
	}

	c.LogMode(false)
	c.DB().SetMaxIdleConns(db.cfg.MaxIdleConns)
	c.DB().SetMaxOpenConns(db.cfg.MaxOpenConns)
	c.DB().SetConnMaxLifetime(time.Second * time.Duration(db.cfg.ConnMaxLifeTime))

	db.DbHandler = c
	return nil
}

func (db *DB) Close() {
	_ = db.DbHandler.Close()
}

func (db *DB) parseConnConfig() string {
	var connHost string
	if db.cfg.Unix != "" {
		connHost = fmt.Sprintf("unix(%s)", db.cfg.Unix)
	} else {
		connHost = fmt.Sprintf("tcp(%s:%d)", db.cfg.Host, db.cfg.Port)
	}
	s := fmt.Sprintf("%s:%s@%s/%s?charset=%s&parseTime=True&loc=Local", db.cfg.User, db.cfg.Pass, connHost, db.cfg.DbName, db.cfg.Charset)
	return s
}
